package com.it.App.utils;

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import java.util.Date;

@Slf4j
@Component
public class JwtUtil {
    private static String header;
    private static String secret;
    private static long expire;

    @Value("${coke.jwt.header}")
    public void setHeader(String header) {
        JwtUtil.header = header;
    }

    @Value("${coke.jwt.secret}")
    public void setSecret(String secret) {
        JwtUtil.secret = secret;
    }

    @Value("${coke.jwt.expire}")
    public void setExpire(long expire) {
        JwtUtil.expire = expire;
    }

    /**
     * 创建JWT Token
     *
     * @param username 用户名
     * @return JWT Token字符串
     */
    public static String createJWT(String username) {
        log.info(secret);
        // 获取当前时间
        Date nowDate = new Date();

        // 计算过期时间，当前时间 + 过期时长
        Date expireDate = new Date(nowDate.getTime() + expire);

        // 使用 JWT Builder 构建 JWT
        return Jwts.builder()
                .setHeaderParam("typ", header) // 设置头部信息，通常为JWT
                .setId(username)
                .setSubject(username) // 设置主题，通常为用户名
                .setIssuedAt(nowDate) // 设置签发时间，即当前时间
                .setExpiration(expireDate) // 设置过期时间
                .signWith(SignatureAlgorithm.HS512, secret) // 使用HS512签名算法和密钥进行签名
                .compact();
    }

    /**
     * 解析JWT Token
     *
     * @param jwt JWT Token字符串
     * @return JWT声明
     */
    public static Claims parseJWT(String jwt) {
        // 这一行将抛出异常，如果它不是已签名的JWS（预期的情况）
        try {
            return Jwts.parser()
                    .setSigningKey(secret)
                    .parseClaimsJws(jwt)
                    .getBody();
        } catch (ExpiredJwtException e) {
            // Token已过期
            return null;
        }
    }

    /**
     * 检查Token是否已过期
     *
     * @return 是否已过期
     */
    public static boolean isTokenExpired(Claims claims) {
        // 检查过期时间是否在当前时间之前
        return claims.getExpiration().before(new Date());
    }

    public static void main(String[] args) {
        // 示例：加密
        String jwt = createJWT("zhangsan");
        log.info(jwt);

        // 示例：解密
        Claims claims = parseJWT(jwt);
        log.info(claims.toString());
        log.info(claims.getId());
    }
}

